package org.fanrui.hrm.exception;

import org.fanrui.hrm.constants.ErrorCode.CustomizeErrorCode;
import org.fanrui.hrm.util.AjaxResult;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.List;

/**
 * 全局异常处理
 *
 * @author fanrui
 */
@RestControllerAdvice//对restController的增强，打了该注解类中的方法，可以在他的controller前后做一些事
public class GlobalExceptionHandler {

    //controller执行前，如果抛出异常，会进入该方法
    //@ExceptionHandler  异常处理
    @ExceptionHandler(value = GlobalException.class)
    public AjaxResult handleGlobalException(GlobalException e) {
        e.printStackTrace();
        return AjaxResult.me().setSuccess(false).setMessage(e.getMessage());
    }

    @ExceptionHandler(value = MethodArgumentNotValidException.class)
    public AjaxResult methodArgumentNotValidException(MethodArgumentNotValidException e) {
        e.printStackTrace();
        StringBuilder sb = new StringBuilder();
        List<ObjectError> errors = e.getBindingResult().getAllErrors();
        for (ObjectError error : errors) {
            sb.append(error.getDefaultMessage()).append(";");
        }
        return AjaxResult.me().setSuccess(false).setMessage(sb.toString());
    }

    @ExceptionHandler(value = Exception.class)
    public AjaxResult handleException(Exception e) {
        e.printStackTrace();
        return AjaxResult.me().setSuccess(false).setMessage(CustomizeErrorCode.SYSTEM_INNER_EXCEPTION.getMessage());
    }
}
